#!/usr/bin/env python
# -*- encoding: utf-8 -*-
'''
@File       :   Main.py    
@Contact    :   LJL959@QQ.com.com
@License    :   (C)Copyright 2019-2020
@Modify Time      @Author    @Version    @Desciption
------------      -------    --------    -----------
2020/4/7 14:41   LiuJiaoLong      1.0         None
@Description :
'''
from Process import ProcessPoems
from Train import GeneratePoem

if __name__ == '__main__':
    words, poem_vector, to_num, x_batches, y_batches = ProcessPoems.ProcessPoems()

    # 用下面这行训练
    GeneratePoem.train(words, poem_vector, x_batches, y_batches)

    # 用下面这行生成古诗，无风格
    # GeneratePoem.gen_poem(words, to_num)

    # 用下面这一行生成
    # GeneratePoem.generate(words, to_num, style_words="狂沙将军战燕然，大漠孤烟黄河骑")

    # 或者用下面这两行生成指定风格
    # style_words_new = input("请输入风格句：")
    # GeneratePoem.generate(words, to_num, style_words_new)
